package com.sun.huazhi.aitaskservice.DTO.response;

import lombok.Data;

import java.util.List;
import java.util.Map;

public class EnhancedImageGenerationResponse {
    private String image_id;
    private String main_image;
    private Map<String, List<Variant>> variants;

    @Data
    public static class Variant {
        private String type;
        private String url;

        public String getType() {
            return type;
        }

        public void setType(String type) {
            this.type = type;
        }

        public String getUrl() {
            return url;
        }

        public void setUrl(String url) {
            this.url = url;
        }
    }

    public String getImage_id() {
        return image_id;
    }

    public void setImage_id(String image_id) {
        this.image_id = image_id;
    }

    public String getMain_image() {
        return main_image;
    }

    public void setMain_image(String main_image) {
        this.main_image = main_image;
    }

    public Map<String, List<Variant>> getVariants() {
        return variants;
    }

    public void setVariants(Map<String, List<Variant>> variants) {
        this.variants = variants;
    }
}
